"""
数据模型模块

定义数据库模型和数据结构。
"""

from .base import Base, BaseModel
from .video import Video, VideoMetadata, VideoClassification
from .task import ProcessingTask, TaskStatus, TaskType
from .project import Project, ProjectVideo
from .jianying_template import JianyingTemplate, JianyingVideoSegment

# 商品相关模型
from .product import Product, ProductStatus, ProductType
from .product_material import (
    ProductRawMaterial, ProductMaterial, ProductMaterialCategory,
    MaterialStatus, MaterialType, QualityLevel
)
from .template import Template, TemplateVideo, TemplateStatus, TemplateType
from .product_output import (
    ProductOutput, ProductOutputMaterial,
    OutputStatus, OutputType
)

__all__ = [
    # 基础模型
    "Base",
    "BaseModel",

    # 视频模型
    "Video",
    "VideoMetadata",
    "VideoClassification",

    # 任务模型
    "ProcessingTask",
    "TaskStatus",
    "TaskType",

    # 项目模型
    "Project",
    "ProjectVideo",

    # 剪映模板模型
    "JianyingTemplate",
    "JianyingVideoSegment",

    # 商品模型
    "Product",
    "ProductStatus",
    "ProductType",

    # 素材模型
    "ProductRawMaterial",
    "ProductMaterial",
    "ProductMaterialCategory",
    "MaterialStatus",
    "MaterialType",
    "QualityLevel",

    # 模板模型
    "Template",
    "TemplateVideo",
    "TemplateStatus",
    "TemplateType",

    # 成品模型
    "ProductOutput",
    "ProductOutputMaterial",
    "OutputStatus",
    "OutputType",
]
